package com.easylinkin.linkappapi.lobar.dto;

import com.easylinkin.linkappapi.lobar.entity.Group;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import lombok.Data;

/**
 * @Author: kan yuanfeng
 * @Date: 2022/5/19 14:18
 * @Description:
 */
@Data
public class GroupDTO extends Group {
  /**
   * 班组长姓名
   */
  private String leadName;

  /**
   * 联系电话
   */
  private String leadPhone;

  /**
   * 班组长姓名
   */
  private int userNum;

  /**
   * 参建单位名称
   * @return
   */
  private String companyName;

  /**
   * 导入模板
   * @return
   */
  public static List<GroupDTO> creatExcel() {
    List<GroupDTO> list = new ArrayList<>();
    GroupDTO groupDTO = new GroupDTO();
    groupDTO.setCompanyName("中国xx有限公司");
    groupDTO.setName("xx班组");
    groupDTO.setServiceArea("施工区,生活区,办公区");
    list.add(groupDTO);
    return list;
  }

  /**
   * 获取通行权限
   */
  public static Map<String,String> getServiceAreaMap(){
    Map<String,String> map = new HashMap<>();
    map.put("施工区","1");
    map.put("办公区","2");
    map.put("生活区","3");
    return map;
  }
}
